home *** CD-ROM | disk | FTP | other *** search
/ SGI Freeware 2002 November / SGI Freeware 2002 November - Disc 2.iso / dist / fw_glimpse.idb / usr / freeware / src / glimpse-3.0 / agrep / recursive.c.z / recursive.c
C/C++ Source or Header  |  1997-09-09  |  3KB  |  173 lines
  1. /* Copyright (c) 1994 Sun Wu, Udi Manber, Burra Gopal.  All Rights Reserved. */
  2. /* The function of the program is to traverse the
  3.    direcctory tree and collect paath names.
  4.    This program is derived from the C-programming language book
  5.    Originally, the program open a directory file as a regular file. But
  6.    it won't work. We have to open a directory file using
  7.    opendir system call, and use readdir() to read each entry of the 
  8.    directory.
  9. */
  10.  
  11. #include "autoconf.h"    /* ../libtemplate/include */
  12. #include <stdio.h>
  13. #include <sys/types.h>
  14. #if    ISO_CHAR_SET
  15. #include <locale.h>
  16. #endif
  17.  
  18. #if HAVE_DIRENT_H
  19. # include <dirent.h>
  20. # define NAMLEN(dirent) strlen((dirent)->d_name)
  21. #else
  22. # define dirent direct
  23. # define NAMLEN(dirent) (dirent)->d_namlen
  24. # if HAVE_SYS_NDIR_H
  25. #  include <sys/ndir.h>
  26. # endif
  27. # if HAVE_SYS_DIR_H
  28. #  include <sys/dir.h>
  29. # endif
  30. # if HAVE_NDIR_H
  31. #  include <ndir.h>
  32. # endif
  33. #endif
  34.  
  35. #include <sys/stat.h>
  36. #include <fcntl.h>
  37. #define BUFSIZE 256
  38. #define DIRSIZE 14
  39. #define max_list 10
  40.  
  41. #ifndef S_ISREG
  42. #define S_ISREG(mode) (0100000&(mode))
  43. #endif
  44.  
  45. #ifndef S_ISDIR
  46. #define S_ISDIR(mode) (0040000&(mode))
  47. #endif
  48.  
  49. char *file_list[max_list*2];
  50. int  fdx=0; /* index of file_List */
  51. extern int Numfiles;
  52. char name_buf[BUFSIZE];
  53.  
  54. void directory();
  55. static void treewalk();
  56.  
  57. /* returns -1 if error, num of matches >= 0 otherwise */
  58. int
  59. recursive(argc, argv)
  60. int argc;
  61. char **argv;
  62. {
  63.     int i,j;
  64.     int num = 0, ret;
  65.  
  66.     for(i=0; i< argc; i++) {
  67.         strcpy(name_buf, argv[i]);
  68.         treewalk(name_buf);
  69.         if(fdx > 0) {
  70.             Numfiles = fdx;
  71.             if ((ret = exec(3, file_list)) == -1) return -1;
  72.             num += ret;
  73.             for(j=0; j<fdx; j++) {
  74.                 free(file_list[j]);
  75.             }
  76.         }
  77.         fdx = 0;
  78.     }
  79.  
  80.     return num;
  81. }
  82.  
  83.  
  84. /*
  85. main(argc, argv)
  86. int argc; char **argv;
  87. {
  88.     char buf[BUFSIZE];
  89.  
  90. #if    ISO_CHAR_SET
  91.     setlocale(LC_ALL, "");
  92. #endif
  93.     if (argc == 1) {
  94.         strcpy(buf, ".");
  95.         treewalk(buf);
  96.     }
  97.     else 
  98.         while(--argc > 0) {
  99.             strcpy(buf, *++argv);
  100.             treewalk(buf);
  101.         }
  102. }
  103. */
  104.  
  105. static void
  106. treewalk(name)
  107. char *name;
  108. {
  109.     struct stat stbuf;
  110.     int i;
  111.     extern void *malloc();
  112.  
  113.     /* printf(" In treewalk\n"); */
  114.     if(lstat(name, &stbuf) == -1) {
  115.         fprintf(stderr, "permission denied or non-existent: %s\n", name);
  116.         return;
  117.     }
  118.     if ((stbuf.st_mode & S_IFMT) == S_IFLNK)  {
  119.         return;
  120.     }
  121.     if (( stbuf.st_mode & S_IFMT) == S_IFDIR) 
  122.         directory(name);
  123.     else {
  124.         file_list[fdx] = (char *)malloc(BUFSIZE);
  125.         strcpy(file_list[fdx++], name);
  126.         /* printf("    %s\n",  name); */
  127.         if(fdx >= max_list) {
  128.             Numfiles = fdx;
  129.             exec(3, file_list);
  130.             for(i=0; i<max_list; i++) free(file_list[i]);
  131.             fdx=0;
  132.         }
  133.  
  134.     }
  135. }
  136.  
  137. void
  138. directory(name)
  139. char *name;
  140. {
  141.     struct dirent *dp;
  142.     char *nbp;
  143.     DIR *dirp;
  144.     /*
  145.         printf("in directory, name= %s\n",name);
  146.     */
  147.     nbp = name + strlen(name);
  148.     if( nbp+DIRSIZE+2 >= name+BUFSIZE ) /* name too long */
  149.     {
  150.         fprintf(stderr, "name too long: %.32s...\n", name);
  151.         return;
  152.     }
  153.     if((dirp = opendir(name)) == NULL) {
  154.         fprintf(stderr, "permission denied: %s\n", name);
  155.         return;
  156.     }
  157.     *nbp++ = '/';
  158.     *nbp = '\0';
  159.     for (dp = readdir(dirp); dp != NULL; dp = readdir(dirp)) {
  160.         if (dp->d_name[0] == '\0' || strcmp(dp->d_name, ".") == 0 || strcmp(dp->d_name, "..")==0) 
  161.             goto CONT;
  162.         /*
  163.             printf("dp->d_name = %s\n", dp->d_name);
  164.         */
  165.         strcpy(nbp, dp->d_name);
  166.         treewalk(name);
  167. CONT:
  168.         ;
  169.     }
  170.     closedir (dirp);
  171.     *--nbp = '\0'; /* restore name */
  172. }
  173.